package network;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.Socket;
import java.net.SocketException;

// UDP 版本的回显服务器
public class UdpEchoServer {
    // 用 socket 对象来抽象表示网卡
    private DatagramSocket socket = null;

    // 使用构造方法来给服务器指定一个端口号
    public UdpEchoServer(int port) throws SocketException {
        socket = new DatagramSocket(port);
    }

    public void start() throws IOException {
        System.out.println("启动服务器!");
        // 用 while 循环来接收客户端请求
        while (true) {
            // 1. 读取客户端发送的请求
            //    使用 receive 方法, 方法的参数是一个输出型参数, 需要事先构造好一个
            //    空白的 DatagramPacket 对象, 交给 receive 填充
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096], 4096);
            socket.receive(requestPacket);

            // DatagramPacket 是特殊对象, 不好直接处理, 把里面的数据拿出来, 构造成一个字符串
            String request = new String(requestPacket.getData(), 0, requestPacket.getLength());

            // 2. 根据请求来计算响应, 此处是一个回显服务器, 请求和响应相同
            String response = process(request);

            // 3. 把响应写会客户端, 使用 send 方法, send 方法的参数也是一个 DatagramPacket,
            //    事先把 DatagramPacket 构造好, 使用响应的数据来构造
            DatagramPacket responsePacket = new DatagramPacket(response.getBytes(),
                    response.getBytes().length,
                    requestPacket.getSocketAddress());
            socket.send(responsePacket);
            // 4. 打印日志, 当前请求的中间结果
            System.out.printf("[%s:%d] req: %s; resp: %s\n", requestPacket.getAddress().toString(),
                    requestPacket.getPort(), request, response);
        }
    }

    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        UdpEchoServer udpEchoServer = new UdpEchoServer(9090);
        udpEchoServer.start();
    }
}
